package com.wc.demo.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import springfox.documentation.spring.web.plugins.Docket;
import springfox.documentation.builders.ApiInfoBuilder;
import springfox.documentation.builders.PathSelectors;
import springfox.documentation.builders.RequestHandlerSelectors;
import springfox.documentation.service.ApiInfo;
import springfox.documentation.service.Contact;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.swagger2.annotations.EnableSwagger2;
import org.springframework.core.env.Environment;
import org.springframework.core.env.Profiles;

import java.util.ArrayList;

/**
 * @author WangChen 2064318526@qq.com
 * @date 2023/8/10 16:09
 */
@Configuration
@EnableSwagger2
public class SwaggerConfig {

    @Bean
    public Docket docket1(Environment environment){
        Profiles profiles = Profiles.of("dev", "test"); // 设置要显示swagger的环境
        boolean isOpen = environment.acceptsProfiles(profiles); // 判断当前是否处于该环境
        return new Docket(DocumentationType.SWAGGER_2)
            // 设置是否启动Swagger，默认为true（不写即可），关闭后Swagger就不生效了
            .enable(isOpen)
            // 分组名称
            .groupName("Demo")
            // 文档信息配置
            .apiInfo(apiInfo())
            // 配置扫描的接口
            .select()
            // 配置扫描哪里的接口
            .apis(RequestHandlerSelectors.basePackage("com.wc.demo.controller"))
            // 过滤请求
            .paths(PathSelectors.any())
            .build();
    }

    @Bean
    public Docket docket2(Environment environment){
        Profiles profiles = Profiles.of("dev", "test"); // 设置要显示swagger的环境
        boolean isOpen = environment.acceptsProfiles(profiles); // 判断当前是否处于该环境
        return new Docket(DocumentationType.SWAGGER_2)
            // 设置是否启动Swagger，默认为true（不写即可），关闭后Swagger就不生效了
            .enable(isOpen)
            // 分组名称
            .groupName("Employee")
            // 文档信息配置
            .apiInfo(apiInfo())
            // 配置扫描的接口
            .select()
            // 配置扫描哪里的接口
            .apis(RequestHandlerSelectors.basePackage("com.wc.demo.controller"))
            // 过滤请求，只扫描请求以/employee开头的接口
            .paths(PathSelectors.ant("/employee/**"))
            .build();
    }

    private ApiInfo apiInfo(){
        return new ApiInfoBuilder()
            .title("WangChen接口文档")//文档标题
            .description("记录项目接口")//文档基本描述
            .contact(new Contact("wangchen", "https://blog.csdn.net/m0_68118199", "2064318526@qq.com"))//联系人信息
            .termsOfServiceUrl("http://terms.service.url/WangChen") // 组织链接
            .version("v1.0")//版本
            .license("Apache 2.0 许可") // 许可
            .licenseUrl("https://www.apache.org/licenses/LICENSE-2.0") // 许可链接
            .extensions(new ArrayList<>()) // 拓展
            .build();
    }
}
